package com.chaosyeb.server.config;

import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.chaosyeb.server.pojo.MailConstants;
import com.chaosyeb.server.pojo.MailLog;
import com.chaosyeb.server.service.IMailLogService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.Binding;
import org.springframework.amqp.core.BindingBuilder;
import org.springframework.amqp.core.DirectExchange;
import org.springframework.amqp.core.Queue;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * @Description : RabbitMQ配置类
 * @Author : Chaos
 * @Date : 2022/1/25 15:53
 * @Version : 1.0
 **/
@Configuration
public class RabbitMQConfig {

    private static final Logger LOGGER = LoggerFactory.getLogger(RabbitMQConfig.class);

    @Autowired
    private CachingConnectionFactory cachingConnectionFactory;
    @Autowired
    private IMailLogService mailLogService;


    /**
     * 配置rabbitMQ模板
     * @return
     */
    @Bean
    public RabbitTemplate rabbitTemplate() {
        RabbitTemplate rabbitTemplate = new RabbitTemplate(cachingConnectionFactory);
        //消息确认回调，确认消息是否到达broker  data：消息唯一标识  ack：确认结果  cause：失败原因
        rabbitTemplate.setConfirmCallback((data, ack, cause) -> {
            String msgId = data.getId();
            if (ack) {
                LOGGER.info("{}==========>消息发送成功", msgId);
                //更新状态status   UpdateWrapper更新
                mailLogService.update(new UpdateWrapper<MailLog>().set("status", 1).eq("msgId",msgId));
            } else {
                LOGGER.error("{}==========>消息发送失败",msgId);
            }
        });
        //消息失败回调 比如router找不到queue时回调 msg:消息主题 repCode：响应码 repText：响应描述 exchange：交换机 路由键
        rabbitTemplate.setReturnCallback((msg, repCode, repText, exchange, routingkey)->{
            LOGGER.error("{}==========>消息发送queue时失败",msg.getBody());
        });
        return rabbitTemplate;
    }


    /**
     * 队列
     * @return
     */
    @Bean
    public Queue queue() {
        return new Queue(MailConstants.MAIL_QUEUE_NAME);
    }

    /**
     * 默认的交换机
     * @return
     */
    @Bean
    public DirectExchange directExchange() {
        return new DirectExchange(MailConstants.MAIL_EXCHANGE_NAME);
    }

    /**
     * 配置绑定信息：
     * @return
     */
    @Bean
    public Binding binding() {
        return BindingBuilder.bind(queue()).to(directExchange()).with(MailConstants.MAIL_EXCHANGE_NAME);
    }
}
